package leetcode.editor.cn.q1_300.q200;
//给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。 
//
// 百度百科中最近公共祖先的定义为：“对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（
//一个节点也可以是它自己的祖先）。” 
//
// 
//
// 示例 1： 
// 
// 
//输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
//输出：3
//解释：节点 5 和节点 1 的最近公共祖先是节点 3 。
// 
//
// 示例 2： 
// 
// 
//输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
//输出：5
//解释：节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身。
// 
//
// 示例 3： 
//
// 
//输入：root = [1,2], p = 1, q = 2
//输出：1
// 
//
// 
//
// 提示： 
//
// 
// 树中节点数目在范围 [2, 10⁵] 内。 
// -10⁹ <= Node.val <= 10⁹ 
// 所有 Node.val 互不相同 。 
// p != q 
// p 和 q 均存在于给定的二叉树中。 
// 
//
// 👍 2315 👎 0

import leetcode.editor.cn.util.tree.btree.TreeNode;

/**
 * 直接复制 235
 */
public class Q236_LowestCommonAncestorOfABinaryTree {
    //leetcode submit region begin(Prohibit modification and deletion)

    class Solution {
        TreeNode p, q;

        public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
            this.p = p;
            this.q = q;
            return process(root).node;
        }

        private subData process(TreeNode node) {
            if (node == null) {
                return new subData();
            }

            subData dataL = process(node.left);
            if (dataL.containP && dataL.containQ) {
                if (dataL.node == null) {
                    dataL.node = node.left;
                }
                return dataL;
            }
            subData dataR = process(node.right);
            if (dataR.containP && dataR.containQ) {
                if (dataR.node == null) {
                    dataR.node = node.left;
                }
                return dataR;
            }

            subData data = new subData();
            if (data.node == null) {
                data.containP = dataL.containP || dataR.containP || node == p;
                data.containQ = dataL.containQ || dataR.containQ || node == q;
                if (data.containQ && data.containP) {
                    data.node = node;
                }
            }
            return data;
        }
    }

    class subData {
        public boolean containP = false;
        public boolean containQ = false;
        public TreeNode node = null;
    }
//leetcode submit region end(Prohibit modification and deletion)


    public static void main(String[] args) {
        Solution solution = new Q236_LowestCommonAncestorOfABinaryTree().new Solution();
        // TO TEST
    }
}